/*
 * Copyright (c) 2018 DISID CORPORATION S.L.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */

package com.bkjk.platform.rabbit;

import org.apache.commons.codec.digest.DigestUtils;
import org.springframework.amqp.core.AmqpAdmin;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.config.RetryInterceptorBuilder;
import org.springframework.amqp.rabbit.config.SimpleRabbitListenerContainerFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.listener.RabbitListenerContainerFactory;
import org.springframework.amqp.rabbit.retry.MessageKeyGenerator;
import org.springframework.amqp.rabbit.retry.MessageRecoverer;
import org.springframework.amqp.rabbit.retry.NewMessageIdentifier;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.util.Assert;

import com.bkjk.platform.rabbit.RabbitProperties.ListenerRetry;
import com.bkjk.platform.rabbit.extend.RepublishMessageRecovererExtend;

/**
 * Configure {@link RabbitListenerContainerFactory} with sensible defaults.
 *
 * @author Stephane Nicoll
 * @author Gary Russell
 * @since 1.3.3
 */
public final class SimpleRabbitListenerContainerFactoryConfigurer {

    private MessageConverter messageConverter;

    private MessageRecoverer messageRecoverer;

    private RabbitProperties rabbitProperties;

    private RabbitTemplate retryRabbitTemplate;

    private AmqpAdmin retryAmqpAdmin;

    /**
     * Configure the specified rabbit listener container factory. The factory can be further tuned and default settings
     * can be overridden.
     * 
     * @param factory the {@link SimpleRabbitListenerContainerFactory} instance to configure
     * @param connectionFactory the {@link ConnectionFactory} to use
     */
    public void configure(SimpleRabbitListenerContainerFactory factory, ConnectionFactory connectionFactory) {
        Assert.notNull(factory, "Factory must not be null");
        Assert.notNull(connectionFactory, "ConnectionFactory must not be null");
        factory.setConnectionFactory(connectionFactory);
        if (this.messageConverter != null) {
            factory.setMessageConverter(this.messageConverter);
        }
        RabbitProperties.AmqpContainer config = this.rabbitProperties.getListener().getSimple();
        factory.setAutoStartup(config.isAutoStartup());
        if (config.getAcknowledgeMode() != null) {
            factory.setAcknowledgeMode(config.getAcknowledgeMode());
        }
        if (config.getConcurrency() != null) {
            factory.setConcurrentConsumers(config.getConcurrency());
        }
        if (config.getMaxConcurrency() != null) {
            factory.setMaxConcurrentConsumers(config.getMaxConcurrency());
        }
        if (config.getPrefetch() != null) {
            factory.setPrefetchCount(config.getPrefetch());
        }
        if (config.getTransactionSize() != null) {
            factory.setTxSize(config.getTransactionSize());
        }
        if (config.getDefaultRequeueRejected() != null) {
            factory.setDefaultRequeueRejected(config.getDefaultRequeueRejected());
        } else {
            factory.setDefaultRequeueRejected(false);
        }
        if (config.getIdleEventInterval() != null) {
            factory.setIdleEventInterval(config.getIdleEventInterval());
        }
        ListenerRetry retryConfig = config.getRetry();
        if (retryConfig.isEnabled()) {
            RetryInterceptorBuilder<?> builder =
                (retryConfig.isStateless() ? RetryInterceptorBuilder.stateless() : RetryInterceptorBuilder.stateful()//
                    .messageKeyGenerator(new MessageKeyGenerator() {

                        @Override
                        public Object getKey(Message message) {
                            String messagId = message.getMessageProperties().getMessageId();
                            if (messagId != null && !messagId.isEmpty())
                                return DigestUtils.md5Hex(message.getBody());
                            else
                                return messagId;
                        }
                    })//
                    .newMessageIdentifier(new NewMessageIdentifier() {

                        @Override
                        public boolean isNew(Message message) {
                            return true;
                        }
                    }));
            builder.maxAttempts(retryConfig.getMaxAttempts());
            builder.backOffOptions(retryConfig.getInitialInterval(), retryConfig.getMultiplier(),
                retryConfig.getMaxInterval());
            MessageRecoverer recoverer;
            if (this.messageConverter != null) {
                recoverer = this.messageRecoverer;
            } else {
                RepublishMessageRecovererExtend republishRecover =
                    new RepublishMessageRecovererExtend(this.retryRabbitTemplate, this.retryAmqpAdmin);
                republishRecover.setInterval(retryConfig.getInitialInterval());
                republishRecover.setRecoverTimes(retryConfig.getMaxAttempts());
                recoverer = republishRecover;
            }
            builder.recoverer(recoverer);
            factory.setAdviceChain(builder.build());
        }

    }

    /**
     * Set the {@link MessageConverter} to use or {@code null} if the out-of-the-box converter should be used.
     * 
     * @param messageConverter the {@link MessageConverter}
     */
    void setMessageConverter(MessageConverter messageConverter) {
        this.messageConverter = messageConverter;
    }

    /**
     * Set the {@link MessageRecoverer} to use or {@code null} to rely on the default.
     * 
     * @param messageRecoverer the {@link MessageRecoverer}
     */
    void setMessageRecoverer(MessageRecoverer messageRecoverer) {
        this.messageRecoverer = messageRecoverer;
    }

    /**
     * Set the {@link RabbitProperties} to use.
     * 
     * @param rabbitProperties the {@link RabbitProperties}
     */
    void setRabbitProperties(RabbitProperties rabbitProperties) {
        this.rabbitProperties = rabbitProperties;
    }

    void setRetryAmqpAdmin(AmqpAdmin amqpAdmin) {
        this.retryAmqpAdmin = amqpAdmin;
    }

    void setRetryRabbitTemplate(RabbitTemplate rabbitTemplate) {
        this.retryRabbitTemplate = rabbitTemplate;
    }

}
